package com.hspedu.newData.dmsxl.动态规划.股票问题;

import org.junit.Test;

/**
 * @Author: yh
 * @description: 
 * @CreateTime: 2025-04-27 
 * @Version: 1.0
 */

public class LC309买卖股票的最佳时机含冷冻期 {

    public int maxProfit(int[] prices) {
        int len = prices.length;

        if (len < 2) return 0;
        int[][] dp = new int[len][3];

        // dp[i][0]  第i天买入的收益
        // dp[i][1]  第i天卖出的收益
        // dp[i][2]  第i天是冷冻期的收益

        dp[0][0] = -prices[0];

        for (int i = 1; i < len; i++) {
            dp[i][0] = Math.max(dp[i-1][0],dp[i-1][2] - prices[i]);
            dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0] + prices[i]);
            dp[i][2] = Math.max(dp[i-1][2],dp[i-1][1]);
        }
        return dp[len - 1][1];
    }

    @Test
    public void test() {
        // 测试代码

        int i = maxProfit(new int[]{1,2,3,4,5,2,3,4,1,21,10,3});
        System.out.println("=========");
    }
}
